package me.spring.cloud.common.components.util.sercurity.encrypt;

import java.security.Key;
import java.security.Security;
import javax.crypto.Cipher;
import lombok.Data;

/**
 * DES加密和解密工具,可以对字符串进行加密和解密操作  。
 */
@Data
public class Des {

  /** 字符串默认键值     */
  private static String strDefaultKey = "national";

  /** 加密工具     */
  private Cipher encryptCipher = null;

  /** 解密工具     */
  private Cipher decryptCipher = null;

  /**
   * 默认构造方法，使用默认密钥
   *
   * @throws Exception
   */
  public Des() throws Exception {
    this(strDefaultKey);
  }

  /**
   * 指定密钥构造方法
   *
   * @param strKey
   *            指定的密钥
   * @throws Exception
   */
  public Des(String strKey) throws Exception {
    Security.addProvider(new com.sun.crypto.provider.SunJCE());
    Key key = getKey(strKey.getBytes());

    encryptCipher = Cipher.getInstance("DES");
    encryptCipher.init(Cipher.ENCRYPT_MODE, key);

    decryptCipher = Cipher.getInstance("DES");
    decryptCipher.init(Cipher.DECRYPT_MODE, key);
  }

  /**
   * 从指定字符串生成密钥，密钥所需的字节数组长度为8位 不足8位时后面补0，超出8位只取前8位
   *
   * @param arrBTmp
   *            构成该字符串的字节数组
   * @return 生成的密钥
   * @throws Exception
   */
  private Key getKey(byte[] arrBTmp) throws Exception {
    // 创建一个空的8位字节数组（默认值为0）
    byte[] arrB = new byte[8];

    // 将原始字节数组转换为8位
    for (int i = 0; i < arrBTmp.length && i < arrB.length; i++) {
      arrB[i] = arrBTmp[i];
    }

    // 生成密钥
    Key key = new javax.crypto.spec.SecretKeySpec(arrB, "DES");

    return key;
  }
}
